package com.ruoyi.wechat.service.impl;

import com.aliyun.oss.ClientException;
import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.OSSException;
import com.aliyun.oss.common.auth.CredentialsProviderFactory;
import com.aliyun.oss.common.auth.EnvironmentVariableCredentialsProvider;
import com.aliyun.oss.model.PutObjectResult;
import com.ruoyi.wechat.service.ISysFileService;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.ByteArrayInputStream;

/**
 * <p>
 * 阿里云对象存储
 * </p>
 *
 * @author YinHeng
 * @since 2024/3/21
 */
@Service
public class OssSysFileServiceImpl implements ISysFileService {

    @Value("${upload.oss.domain}")
    private String domain;
    @Value("${upload.oss.bucketName}")
    private String bucketName;
    @Value("${upload.oss.accessKeyId}")
    private String accessKeyId;
    @Value("${upload.oss.accessSecret}")
    private String accessSecret;


    @Override
    public String uploadFile(String dicrt, MultipartFile file) throws Exception {
        // 获取文件后缀名
        String fileExtension = getFileExtension(file);

        // 获取文件名
        String fileName = getFileName(file);
        String path = dicrt + "/" + fileName + "." + fileExtension;

        // Endpoint以华东1（杭州）为例，其它Region请按实际情况填写。
        String scheme = "https://";
        String endpoint = scheme + domain;

        // 从环境变量中获取访问凭证。运行本代码示例之前，请确保已设置环境变量OSS_ACCESS_KEY_ID和OSS_ACCESS_KEY_SECRET。
        EnvironmentVariableCredentialsProvider credentialsProvider = CredentialsProviderFactory.newEnvironmentVariableCredentialsProvider();

        // 填写Bucket名称，例如examplebucket。
        // 填写Object完整路径，例如exampledir/exampleobject.txt。Object完整路径中不能包含Bucket名称。
        String fileUrl = "https://" + bucketName + "." + domain + "/" + path;

        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessSecret);
        PutObjectResult putObjectResult = null;
        try {
            putObjectResult = ossClient.putObject(bucketName, path, new ByteArrayInputStream(file.getBytes()));
            return fileUrl;
        } catch (OSSException oe) {
            System.out.println("Caught an OSSException, which means your request made it to OSS, "
                    + "but was rejected with an error response for some reason.");
            System.out.println("Error Message:" + oe.getErrorMessage());
            System.out.println("Error Code:" + oe.getErrorCode());
            System.out.println("Request ID:" + oe.getRequestId());
            System.out.println("Host ID:" + oe.getHostId());
        } catch (ClientException ce) {
            System.out.println("Caught an ClientException, which means the client encountered "
                    + "a serious internal problem while trying to communicate with OSS, "
                    + "such as not being able to access the network.");
            System.out.println("Error Message:" + ce.getMessage());
        } finally {
            if (ossClient != null) {
                ossClient.shutdown();
            }
        }
        return fileUrl;
    }


    // 获取文件后缀名
    private static String getFileExtension(MultipartFile file) {
        String originalFileName = file.getOriginalFilename();
        if (originalFileName != null) {
            return originalFileName.substring(originalFileName.lastIndexOf(".") + 1);
        }
        return null;
    }

    // 获取文件名（不带后缀）
    private static String getFileName(MultipartFile file) {
        String originalFileName = file.getOriginalFilename();
        if (originalFileName != null) {
            if (originalFileName.lastIndexOf(".") != -1) {
                return originalFileName.substring(0, originalFileName.lastIndexOf("."));
            } else {
                return originalFileName;
            }
        }
        return null;
    }

}
